# YOLOv5 requirements
# Usage: pip install -r requirements.txt

# Base ------------------------------------------------------------------------
gitpython>=3.1.30
matplotlib>=3.3
#numpy>=1.23.5 #
numpy>=1.19.5 # FOR LEGACY SYSTEMS ONLY
opencv-python>=4.1.1
pillow>=10.3.0
psutil  # system resources
PyYAML>=5.3.1
requests>=2.32.2
scipy>=1.4.1
thop>=0.1.1  # FLOPs computation
torch>=1.8.0  # see https://pytorch.org/get-started/locally (recommended)
torchvision>=0.9.0
tqdm>=4.66.3
ultralytics>=8.2.64  # https://ultralytics.com
protobuf<=3.20.1  # https://github.com/ultralytics/yolov5/issues/8012

wandb
tqdm
tensorflow>=2.4.1
tensorflowjs>=3.9.0
onnx>=1.9.0
onnx-simplifier
scikit-learn
coremltools

# Logging ---------------------------------------------------------------------
 tensorboard>=2.4.1
 clearml>=1.2.0
 comet

# Plotting --------------------------------------------------------------------
pandas>=1.1.4
seaborn>=0.11.0

# Export ----------------------------------------------------------------------
 coremltools>=6.0  # CoreML export
 onnx>=1.10.0  # ONNX export
 onnx-simplifier>=0.4.1  # ONNX simplifier
 nvidia-pyindex  # TensorRT export
 nvidia-tensorrt  # TensorRT export
 scikit-learn<=1.1.2  # CoreML quantization
 tensorflow>=2.4.0,<=2.13.1  # TF exports (-cpu, -aarch64, -macos)
 tensorflowjs>=3.9.0  # TF.js export
 openvino-dev>=2023.0  # OpenVINO export

# Deploy ----------------------------------------------------------------------
packaging  # Migration of deprecated pkg_resources packages
setuptools>=70.0.0 # Snyk vulnerability fix
# tritonclient[all]~=2.24.0

# Extras ----------------------------------------------------------------------
# ipython  # interactive notebook
# mss  # screenshots
# albumentations>=1.0.3
# pycocotools>=2.0.6  # COCO mAP
urllib3>=2.6.0 ; python_version > "3.8" # not directly required, pinned by Snyk to avoid a vulnerability
